﻿using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using System.Runtime.InteropServices;
using System.Text;

namespace nwrole.lib
{
    public class ServiceUtil
    {
        #region External
        [DllImport("advapi32.dll")]
        public static extern int StartService (IntPtr SVHANDLE, int dwNumServiceArgs, string lpServiceArgVectors);

        [DllImport("advapi32.dll")]
        public static extern IntPtr OpenSCManager (string lpMachineName, string lpSCDB, int scParameter);
        [DllImport("Advapi32.dll")]
        public static extern IntPtr CreateService (IntPtr SC_HANDLE, string lpSvcName, string lpDisplayName,
        int dwDesiredAccess, int dwServiceType, int dwStartType, int dwErrorControl, string lpPathName,
        string lpLoadOrderGroup, int lpdwTagId, string lpDependencies, string lpServiceStartName, string lpPassword);
        [DllImport("advapi32.dll")]
        public static extern void CloseServiceHandle (IntPtr SCHANDLE);
        [DllImport("advapi32.dll", SetLastError = true)]
        public static extern IntPtr OpenService (IntPtr SCHANDLE, string lpSvcName, int dwNumServiceArgs);
        [DllImport("advapi32.dll")]
        public static extern int DeleteService (IntPtr SVHANDLE);
        [DllImport("advapi32.dll")]
        public static extern bool ChangeServiceConfig2 (IntPtr hService, InfoLevel dwInfoLevel, [MarshalAs(UnmanagedType.Struct)] ref SERVICE_DESCRIPTION lpInfo);
        [DllImport("kernel32.dll")]
        public static extern int GetLastError ();
        [DllImport("kernel32.dll")]
        private static extern int FormatMessage (
                                              int dwFlags,
                                              IntPtr lpSource,
                                              int dwMessageId,
                                              int dwLanguageId,
                                              out StringBuilder MsgBuffer,
                                              int nSize,
                                              IntPtr lpParams);

        public enum InfoLevel : int
        {
            SERVICE_CONFIG_DESCRIPTION = 1,
            SERVICE_CONFIG_FAILURE_ACTIONS = 2
        }

        [StructLayout(LayoutKind.Sequential)]
        public struct SERVICE_DESCRIPTION
        {
            public string lpDescription;
        }
        #endregion

        public static bool UninstallService (string svcName)
        {
            int SC_MANAGER_CREATE_SERVICE = 0x0002;
            IntPtr sc_handle = OpenSCManager(null, null, SC_MANAGER_CREATE_SERVICE);
            if (sc_handle.ToInt32() != 0)
            {
                IntPtr svc_hndl = OpenService(sc_handle, svcName, 0xF01FF);
                if (svc_hndl.ToInt32() != 0)
                {
                    int i = DeleteService(svc_hndl);
                    if (i != 0)
                    {
                        CloseServiceHandle(svc_hndl);
                        svc_hndl = System.IntPtr.Zero;
                        svc_hndl = System.IntPtr.Zero;
                    }
                }
                CloseServiceHandle(sc_handle);
            }
            return true;
        }

        public static bool StartService (string svcName)
        {
            int SC_MANAGER_CREATE_SERVICE = 0x0002;
            IntPtr sc_handle = OpenSCManager(null, null, SC_MANAGER_CREATE_SERVICE);
            if (sc_handle.ToInt32() != 0)
            {
                IntPtr svc_hndl = OpenService(sc_handle, svcName, 0xF01FF);
                if (svc_hndl.ToInt32() != 0)
                {
                    int i = StartService(svc_hndl, 0, null);
                    /*if (i != 0)
                    {
                        int a = 1;
                    }
                    else
                    {
                        int b = 1;
                    }*/
                }
                CloseServiceHandle(svc_hndl);
                CloseServiceHandle(sc_handle);
            }
            return true;
        }

        public static bool InstallService (string svcPath, string svcName, string svcDispName, string svcStartName, string svcPassword)
        {
            int SC_MANAGER_CREATE_SERVICE = 0x0002;
            int SERVICE_WIN32_OWN_PROCESS = 0x00000010;
            //int SERVICE_DEMAND_START = 0x00000003;
            int SERVICE_ERROR_NORMAL = 0x00000001;
            int STANDARD_RIGHTS_REQUIRED = 0xF0000;
            int SERVICE_QUERY_CONFIG = 0x0001;
            int SERVICE_CHANGE_CONFIG = 0x0002;
            int SERVICE_QUERY_STATUS = 0x0004;
            int SERVICE_ENUMERATE_DEPENDENTS = 0x0008;
            int SERVICE_START = 0x0010;
            int SERVICE_STOP = 0x0020;
            int SERVICE_PAUSE_CONTINUE = 0x0040;
            int SERVICE_INTERROGATE = 0x0080;
            int SERVICE_USER_DEFINED_CONTROL = 0x0100;
            int SERVICE_ALL_ACCESS = (STANDARD_RIGHTS_REQUIRED |
            SERVICE_QUERY_CONFIG |
            SERVICE_CHANGE_CONFIG |
            SERVICE_QUERY_STATUS |
            SERVICE_ENUMERATE_DEPENDENTS |
            SERVICE_START |
            SERVICE_STOP |
            SERVICE_PAUSE_CONTINUE |
            SERVICE_INTERROGATE |
            SERVICE_USER_DEFINED_CONTROL);
            int SERVICE_AUTO_START = 0x00000002;

            try
            {
                IntPtr sc_handle = OpenSCManager(null, null, SC_MANAGER_CREATE_SERVICE);
                if (sc_handle.ToInt32() != 0)
                {
                    IntPtr sv_handle = CreateService(sc_handle, svcName, svcDispName, SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, svcPath,
                                       null, 0, null, svcStartName, svcPassword);
                    if (sv_handle.ToInt32() == 0)
                    {
                        // sorry, we failed
                        int err = GetLastError();
                        int dwFormatFlags = 0x00000100 | 0x00001000;
                        int dwLanguageId = 0x0800;
                        StringBuilder pMessageBuffer = new StringBuilder();
                        int dwBufferLength;
                        dwBufferLength = FormatMessage(dwFormatFlags, IntPtr.Zero, err, dwLanguageId, out pMessageBuffer, 8096, IntPtr.Zero);
                        if (pMessageBuffer.Length > 0)
                        {
                            throw new Exception(pMessageBuffer.ToString());
                        }
                        CloseServiceHandle(sc_handle);

                        EventLogger.LogInformation("FEL: " + err);
                        return false;
                    }
                    else
                    {
                        SERVICE_DESCRIPTION ServiceDescription;
                        ServiceDescription.lpDescription = "";
                        ChangeServiceConfig2(sv_handle, InfoLevel.SERVICE_CONFIG_DESCRIPTION, ref ServiceDescription);
                        CloseServiceHandle(sv_handle);
                        CloseServiceHandle(sc_handle);
                        return true;
                    }
                }
                else
                {
                  int err = GetLastError();
                  EventLogger.LogInformation("SCM not opened successfully - " + err);
                    //Console.WriteLine("SCM not opened successfully");
                    return false;
                }
            }
            catch (Exception e)
            {
                throw e;
            }
        }
    }
}
